package com.atguigu.auth.controller;

import com.atguigu.auth.service.SysMenuService;
import com.atguigu.auth.service.SysUserService;
import com.atguigu.common.config.exception.GuiGuException;
import com.atguigu.common.jwt.JwtHelper;
import com.atguigu.common.result.Result;
import com.atguigu.common.utils.MD5;
import com.atguigu.model.system.SysMenu;
import com.atguigu.model.system.SysUser;
import com.atguigu.vo.system.LoginVo;
import com.atguigu.vo.system.RouterVo;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import io.swagger.annotations.Api;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * ClassName: IndexController
 * Package: com.atguigu.auth.controller
 * Description:
 *
 * @Author 朱安迪
 * @Create 2023/4/26 9:39
 * @Version 1.0
 */
@Api(tags = "后台登录管理")
@RestController
@RequestMapping("/admin/system/index")
public class IndexController {
    @Autowired
    private SysUserService sysUserService;

    @Autowired
    private SysMenuService sysMenuService;

    //login
    @PostMapping("/login")
    public Result login(@RequestBody LoginVo loginVo){

//        //{code: 200, data: {token: "admin-token"}}
//        Map<String,Object> map  = new HashMap<>();
//        map.put("token","admin-token");
//        return Result.ok(map);
        //1. 获取输入用户名和密码
        //2. 根据用户名查询数据库
        String username = loginVo.getUsername();
        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<SysUser>();
        wrapper.eq(SysUser::getUsername,username);
        SysUser sysUser = sysUserService.getOne(wrapper);

        //3. 用户信息是否存在
        if(sysUser == null){
            throw new GuiGuException(201,"用户不存在");
        }

        //4. 判断密码
        //数据库密码
        String password_db = sysUser.getPassword();
        //获取输入的密码
        String password_input = MD5.encrypt(loginVo.getPassword());
        if(!password_db.equals(password_input)){
            throw new GuiGuException(201,"密码输入错误");
        }

        //5. 判断用户是否被禁用
        if(sysUser.getStatus().intValue() == 0){
            throw new GuiGuException(201,"用户被禁用，请联系管理员");
        }

        //6. 使用jwt根据userId和username生成token字符串
        String token = JwtHelper.createToken(sysUser.getId(), sysUser.getUsername());

        //7. 返回token
        Map<String,Object> map = new HashMap<>();
        map.put("token",token);
        return Result.ok(map);
    }

    //info
    @GetMapping("/info")
    public Result info(HttpServletRequest request){
        //1. 从请求头获取用户信息（获取请求头token字符串）
        String token = request.getHeader("token");
        //2. 从token字符串获取userId 或者 username
        Long userId = JwtHelper.getUserId(token);
        //3. 根据用户id查询数据库，把用户信息获取出来
        SysUser sysUser = sysUserService.getById(userId);
        //4. 根据userId获取用户可以操作菜单列表（粗粒度），动态构建路由结构
        List<RouterVo> routerList = sysMenuService.findUserMenuListByUserId(userId);
        //5. 根据userId获取用户可以操作菜单按钮（细粒度）
        List<String> permsList = sysMenuService.findUserPermsByUserId(userId);
        //6. 返回相应数据
        Map<String,Object> map = new HashMap<>();
        map.put("roles","[admin]");
        map.put("name",sysUser.getName());
        map.put("avatar","https://wpimg.wallstcn.com/f778738c-e4f8-4870-b634-56703b4acafe.gif");
        //TODO 返回用户可以操作菜单
        map.put("routers",routerList);
        //TODO 返回用户可以操作按钮
        map.put("buttons",permsList);
        return Result.ok(map);
    }

    //logout
    @PostMapping("/logout")
    public Result logout(){
        //{code: 20000, data: "success"}
        return Result.ok("success");
    }

}
